package com.aspire.aimini.repository;

import com.aspire.aimini.model.entity.Material;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;

import java.time.LocalDateTime;
import java.util.List;
import java.util.Map;

/**
 * 素材Repository接口
 *
 * @author wangguiyou
 * @date 2024-01-01
 * @description 素材数据访问层接口
 */
@Mapper
public interface MaterialRepository extends BaseMapper<Material> {

    /**
     * 分页查询素材列表
     *
     * @param page       分页对象
     * @param userId     用户ID
     * @param keyword    搜索关键词
     * @param tags       标签筛选
     * @param sourceType 来源类型
     * @param fileType   文件类型
     * @param isFavorite 是否收藏
     * @param isArchived 是否归档
     * @param sortBy     排序字段
     * @param sortOrder  排序方向
     * @param startTime  开始时间
     * @param endTime    结束时间
     * @return 分页结果
     */
    IPage<Material> selectMaterialPage(
            Page<Material> page,
            @Param("userId") Long userId,
            @Param("keyword") String keyword,
            @Param("tags") List<String> tags,
            @Param("sourceType") String sourceType,
            @Param("fileType") String fileType,
            @Param("isFavorite") Boolean isFavorite,
            @Param("isArchived") Boolean isArchived,
            @Param("sortBy") String sortBy,
            @Param("sortOrder") String sortOrder,
            @Param("startTime") LocalDateTime startTime,
            @Param("endTime") LocalDateTime endTime);

    /**
     * 根据标签查询素材列表
     *
     * @param tagIds 标签ID列表
     * @param userId 用户ID
     * @return 素材列表
     */
    List<Material> selectByTagIds(@Param("tagIds") List<Long> tagIds, @Param("userId") Long userId);

    /**
     * 更新素材查看次数
     *
     * @param id 素材ID
     */
    void incrementViewCount(@Param("id") Long id);

    /**
     * 更新素材下载次数
     *
     * @param id 素材ID
     */
    void incrementDownloadCount(@Param("id") Long id);

    /**
     * 批量更新素材状态
     *
     * @param ids    素材ID列表
     * @param status 状态
     * @param userId 用户ID
     * @return 更新数量
     */
    int batchUpdateStatus(@Param("ids") List<Long> ids, @Param("status") Integer status, @Param("userId") Long userId);

    /**
     * 批量删除素材
     *
     * @param ids    素材ID列表
     * @param userId 用户ID
     * @return 删除数量
     */
    int batchDelete(@Param("ids") List<Long> ids, @Param("userId") Long userId);

    /**
     * 统计用户素材数量
     *
     * @param userId 用户ID
     * @return 素材数量
     */
    Long countByUserId(@Param("userId") Long userId);

    /**
     * 统计用户存储使用量
     *
     * @param userId 用户ID
     * @return 存储使用量(字节)
     */
    Long sumStorageByUserId(@Param("userId") Long userId);

    /**
     * 查询用户收藏素材列表
     *
     * @param page   分页对象
     * @param userId 用户ID
     * @return 分页结果
     */
    IPage<Material> selectFavoritesByUserId(Page<Material> page, @Param("userId") Long userId);

    /**
     * 查询最近上传的素材
     *
     * @param userId 用户ID
     * @param limit  限制数量
     * @return 素材列表
     */
    List<Material> selectRecentUploads(@Param("userId") Long userId, @Param("limit") Integer limit);

    /**
     * 根据文件类型统计素材数量
     *
     * @param userId 用户ID
     * @return 统计结果
     */
    List<Map<String, Object>> countByFileType(@Param("userId") Long userId);

    /**
     * 根据来源类型统计素材数量
     *
     * @param userId 用户ID
     * @return 统计结果
     */
    List<Map<String, Object>> countBySourceType(@Param("userId") Long userId);

    /**
     * 查询相似素材(根据标签)
     *
     * @param materialId 素材ID
     * @param userId     用户ID
     * @param limit      限制数量
     * @return 素材列表
     */
    List<Material> selectSimilarMaterials(@Param("materialId") Long materialId, @Param("userId") Long userId,
            @Param("limit") Integer limit);

    /**
     * 清理过期的临时文件
     *
     * @param expiredTime 过期时间
     * @return 清理数量
     */
    int cleanExpiredTempFiles(@Param("expiredTime") LocalDateTime expiredTime);
}